# Copyright (c) 2025 Huawei Technologies Co., Ltd.
# This file is a part of the CANN Open Software.
# Licensed under CANN Open Software License Agreement Version 1.0 (the "License").
# Please refer to the License for details. You may not use this file except in compliance with the License.
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
# EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
# MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
# See LICENSE in the root of the software repository for the full text of the License.


curDirName=$(basename "$(pwd)")
deviceId=$1
targetName=$2
prof_func=$3


cd ..
if [[ ${prof_func} == "0" ]]; then
    ./batchgemm_profiling_make.sh $targetName
elif [[ ${prof_func} == "1" ]]; then
    ./batchgemm_make.sh $targetName
fi


cd $curDirName

if [[ ${prof_func} == "0" ]]; then
    python3 batch_prof.py $deviceId $targetName
elif [[ ${prof_func} == "1" ]]; then
    python3 batch_func.py $deviceId $targetName
fi
